[ROCm] Skip unsupported unit tests on MI200 by amd-jianli12 · Pull Request #40369 · openxla/xla

amd-jianli12 · 2026-04-03T17:00:51Z

Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt.

📝 Summary of Changes
hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests.

🎯 Justification
This change is to make the following tests pass on MI200 GPU:
//xla/service/gpu:dot_algorithm_support_test_amdgpu_any
//xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any

🚀 Kind of Contribution
🐛 Bug Fix

@xla-rotation Would you kindly have a look please?

Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt.

i-chaochen · 2026-04-04T12:21:09Z

I have skipped MOSAIC GPU test from acccb3d as it's NV-only backend.

i-chaochen · 2026-04-04T16:56:33Z

If we can bump up ROCm7.2.1 #40405 CI ROCm JAX will be green as well.

And CI / XLA Linux X86 GPU ONEAPI is irrelevent to this PR.

i-chaochen · 2026-04-06T10:32:35Z

Hi @dimitar-asenov could you review this PR first please? once this is merged we can have a green XLA CI and then we can bazel-cache and CI will be much faster.

Next one is #40405 for JAX UT on XLA

akuegel · 2026-04-07T06:57:49Z


 TEST_F(AlgorithmTest, Algorithm6xBF16) {
+  if (GpuComputeComp().IsRocm()) {
+    if (GpuComputeComp().rocm_compute_capability()->gfx9_mi200())


nit: the style guide says we should put '{' '}' around if blocks (even if it is just a single line).

Also, why not just make it a && condition together with the previous if?

Yes, I should. Will update.

@xla-rotation

Imported from GitHub PR #40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ce by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f4 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=#40369 from ROCm:ci_upstream_ut_fix b632dc5 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR #40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ce by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f4 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=#40369 from ROCm:ci_upstream_ut_fix b632dc5 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR #40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ce by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f4 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=#40369 from ROCm:ci_upstream_ut_fix b632dc5 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR #40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ce by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f4 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=#40369 from ROCm:ci_upstream_ut_fix b632dc5 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR #40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ce by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f4 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=#40369 from ROCm:ci_upstream_ut_fix b632dc5 PiperOrigin-RevId: 895851209

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 895851209

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 896242389

@xla-rotation

Imported from GitHub PR openxla/xla#40369 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. 📝 Summary of Changes hipBlasLt is returning empty algorithms for multi-pass bf16_bf16_f32 dot configuration on MI200, so disable these tests. 🎯 Justification This change is to make the following tests pass on MI200 GPU: //xla/service/gpu:dot_algorithm_support_test_amdgpu_any //xla/backends/gpu/codegen/triton:dot_algorithms_test_amdgpu_any 🚀 Kind of Contribution 🐛 Bug Fix @xla-rotation Would you kindly have a look please? Copybara import of the project: -- 82352ceeca14167b9f024edc9f7ea5e4e1dc62c5 by Jian Li <Jian.Li@amd.com>: [ROCm] Skip unsupported unit tests on MI200 Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt. -- e6f64f487a59729525713b73369187b5de765554 by Chao Chen <cchen104@amd.com>: skipped Mosaic GPU test as it's NV-only -- b632dc5267cacc1ae512727c281acdf1729508b9 by Jian Li <Jian.Li@amd.com>: Address review: Enforce style compliance and flatten conditionals Merging this change closes #40369 PiperOrigin-RevId: 896279515

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 896242242

[ROCm] Skip unsupported unit tests on MI200

82352ce

Skip multi-pass dot tests for BF16 on MI200 since there's no valid algorithm being returned by hipBlasLt.

i-chaochen requested a review from dextero April 3, 2026 21:33

i-chaochen added the AMD-GPU XLA on AMD GPU label Apr 3, 2026

github-project-automation bot added this to OpenXLA:GPU - AMD/ROCm Apr 3, 2026

github-project-automation bot moved this to Pending Google Review in OpenXLA:GPU - AMD/ROCm Apr 3, 2026

i-chaochen added 2 commits April 3, 2026 21:42

trigger ci

d427439

skipped Mosaic GPU test as it's NV-only

e6f64f4

i-chaochen requested review from beckerhe and dimitar-asenov April 4, 2026 12:23

i-chaochen mentioned this pull request Apr 4, 2026

[ROCm] bump up ROCm CI to rocm7.2.1 #40405

Closed

akuegel reviewed Apr 7, 2026

View reviewed changes

Address review: Enforce style compliance and flatten conditionals

b632dc5

akuegel approved these changes Apr 7, 2026

View reviewed changes

copybara-service bot mentioned this pull request Apr 7, 2026

PR #40369: [ROCm] Skip unsupported unit tests on MI200 #40475

Merged

copybara-service bot mentioned this pull request Apr 7, 2026

PR #40369: [ROCm] Skip unsupported unit tests on MI200 tensorflow/tensorflow#115347

Merged

beckerhe approved these changes Apr 8, 2026

View reviewed changes

copybara-service bot closed this in d194a36 Apr 8, 2026

github-project-automation bot moved this from Pending Google Review to Done in OpenXLA:GPU - AMD/ROCm Apr 8, 2026

copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Apr 8, 2026

Automated Code Change

3649658

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 896242389

copybara-service bot mentioned this pull request Apr 8, 2026

Automated Code Change tensorflow/tensorflow#115406

Draft

copybara-service bot pushed a commit to tensorflow/tensorflow that referenced this pull request Apr 8, 2026

Automated Code Change

2b74ef8

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#40369 from ROCm:ci_upstream_ut_fix b632dc5267cacc1ae512727c281acdf1729508b9 PiperOrigin-RevId: 896242242

copybara-service bot mentioned this pull request Apr 8, 2026

Automated Code Change tensorflow/tensorflow#115407

Draft

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[ROCm] Skip unsupported unit tests on MI200#40369

[ROCm] Skip unsupported unit tests on MI200#40369
amd-jianli12 wants to merge 4 commits intoopenxla:mainfrom
ROCm:ci_upstream_ut_fix

amd-jianli12 commented Apr 3, 2026 •

edited

Loading

Uh oh!

i-chaochen commented Apr 4, 2026

Uh oh!

i-chaochen commented Apr 4, 2026

Uh oh!

i-chaochen commented Apr 6, 2026 •

edited

Loading

Uh oh!

akuegel Apr 7, 2026

Uh oh!

amd-jianli12 Apr 7, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

Conversation

amd-jianli12 commented Apr 3, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

i-chaochen commented Apr 4, 2026

Uh oh!

i-chaochen commented Apr 4, 2026

Uh oh!

i-chaochen commented Apr 6, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

akuegel Apr 7, 2026

Choose a reason for hiding this comment

Uh oh!

amd-jianli12 Apr 7, 2026

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

amd-jianli12 commented Apr 3, 2026 •

edited

Loading

i-chaochen commented Apr 6, 2026 •

edited

Loading